package cn.iocoder.yudao.module.ems.service.classification;

import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
import cn.iocoder.yudao.framework.common.util.collection.CollectionUtils;
import cn.iocoder.yudao.framework.tenant.core.aop.TenantIgnore;
import cn.iocoder.yudao.module.ems.mq.producer.ClassificationProducer;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableMultimap;
import com.google.common.collect.Multimap;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Lazy;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;

import java.util.*;
import cn.iocoder.yudao.module.ems.controller.admin.classification.vo.*;
import cn.iocoder.yudao.module.ems.dal.dataobject.classification.ClassificationDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;

import cn.iocoder.yudao.module.ems.convert.classification.ClassificationConvert;
import cn.iocoder.yudao.module.ems.dal.mysql.classification.ClassificationMapper;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.ems.enums.ErrorCodeConstants.*;

/**
 * 分类 Service 实现类
 *
 * @author dason
 */
@Service
@Validated
@Slf4j
public class ClassificationServiceImpl implements ClassificationService {

    /**
            * 定时执行 {@link #schedulePeriodicRefresh()} 的周期
     * 因为已经通过 Redis Pub/Sub 机制，所以频率不需要高
     */
    private static final long SCHEDULER_PERIOD = 5 * 60 * 1000L;

    /**
     * 部门缓存
     * key：部门编号 {@link ClassificationDO#getId()}
     *
     * 这里声明 volatile 修饰的原因是，每次刷新时，直接修改指向
     */
    @SuppressWarnings("FieldCanBeLocal")
    private volatile Map<Long, ClassificationDO> classificationCache;
    /**
     * 父部门缓存
     * key：部门编号 {@link ClassificationDO#getParentId()}
     * value: 直接子部门列表
     *
     * 这里声明 volatile 修饰的原因是，每次刷新时，直接修改指向
     */
    private volatile Multimap<Long, ClassificationDO> parentClassificationCache;
    /**
     * 缓存部门的最大更新时间，用于后续的增量轮询，判断是否有更新
     */
    private volatile Date maxUpdateTime;

    @Resource
    private ClassificationMapper classificationMapper;

    @Resource
    private ClassificationProducer classificationProducer;

    @Resource
    @Lazy // 注入自己，所以延迟加载
    private ClassificationService self;

    public ClassificationServiceImpl() {
    }

    @Override
    @PostConstruct
    @TenantIgnore // 初始化缓存，无需租户过滤
    public synchronized void initLocalCache() {
        // 获取部门列表，如果有更新
        List<ClassificationDO> classificationList = loadClassificationIfUpdate(maxUpdateTime);
        if (CollUtil.isEmpty(classificationList)) {
            return;
        }

        // 构建缓存
        ImmutableMap.Builder<Long, ClassificationDO> builder = ImmutableMap.builder();
        ImmutableMultimap.Builder<Long, ClassificationDO> parentBuilder = ImmutableMultimap.builder();
        classificationList.forEach(sysRoleDO -> {
            builder.put(sysRoleDO.getId(), sysRoleDO);
            parentBuilder.put(sysRoleDO.getParentId(), sysRoleDO);
        });
        // 设置缓存
        classificationCache = builder.build();
        parentClassificationCache = parentBuilder.build();
        maxUpdateTime = CollectionUtils.getMaxValue(classificationList, ClassificationDO::getUpdateTime);
        log.info("[initLocalCache][初始化 Classification 数量为 {}]", classificationList.size());
    }

    /**
     * 如果部门发生变化，从数据库中获取最新的全量部门。
     * 如果未发生变化，则返回空
     *
     * @param maxUpdateTime 当前部门的最大更新时间
     * @return 部门列表
     */
    protected List<ClassificationDO> loadClassificationIfUpdate(Date maxUpdateTime) {
        // 第一步，判断是否要更新。
        if (maxUpdateTime == null) { // 如果更新时间为空，说明 DB 一定有新数据
            log.info("[loadMenuIfUpdate][首次加载全量部门]");
        } else { // 判断数据库中是否有更新的部门
            if (classificationMapper.selectCountByUpdateTimeGt(maxUpdateTime) == 0) {
                return null;
            }
            log.info("[loadMenuIfUpdate][增量加载全量部门]");
        }
        // 第二步，如果有更新，则从数据库加载所有部门
        return classificationMapper.selectList();
    }

    @Scheduled(fixedDelay = SCHEDULER_PERIOD, initialDelay = SCHEDULER_PERIOD)
    public void schedulePeriodicRefresh() {
        self.initLocalCache();
    }

    @Override
    public Long createClassification(ClassificationCreateReqVO createReqVO) {
        // 插入
        ClassificationDO classification = ClassificationConvert.INSTANCE.convert(createReqVO);
        classificationMapper.insert(classification);
        // 发送刷新消息
        classificationProducer.sendClassificationRefreshMessage();
        // 返回
        return classification.getId();
    }

    @Override
    public void updateClassification(ClassificationUpdateReqVO updateReqVO) {
        // 校验存在
        this.validateClassificationExists(updateReqVO.getId());
        // 更新
        ClassificationDO updateObj = ClassificationConvert.INSTANCE.convert(updateReqVO);
        classificationMapper.updateById(updateObj);
        // 发送刷新消息
        classificationProducer.sendClassificationRefreshMessage();
    }

    @Override
    public void deleteClassification(Long id) {
        // 校验存在
        this.validateClassificationExists(id);
        // 删除
        classificationMapper.deleteById(id);
        // 发送刷新消息
        classificationProducer.sendClassificationRefreshMessage();
    }

    @Override
    public List<ClassificationDO> getSimpleClassifications(ClassificationListReqVO reqVO) {
        return classificationMapper.selectList(reqVO);
    }

    @Override
    public List<ClassificationDO> getClassifications(Collection<Long> ids) {
        return classificationMapper.selectBatchIds(ids);
    }

    private void validateClassificationExists(Long id) {
        if (classificationMapper.selectById(id) == null) {
            throw exception(CLASSIFICATION_NOT_EXISTS);
        }
    }

    @Override
    public ClassificationDO getClassification(Long id) {
        return classificationMapper.selectById(id);
    }

    @Override
    public void validClassifications(Collection<Long> ids) {
        if (CollUtil.isEmpty(ids)) {
            return;
        }
        // 获得科室信息
        List<ClassificationDO> classifications = classificationMapper.selectBatchIds(ids);
        Map<Long, ClassificationDO> classificationMap = CollectionUtils.convertMap(classifications, ClassificationDO::getId);
        // 校验
        ids.forEach(id -> {
            ClassificationDO classification = classificationMap.get(id);
            if (classification == null) {
                throw exception(CLASSIFICATION_NOT_EXISTS);
            }
            if (!CommonStatusEnum.ENABLE.getStatus().equals(classification.getStatus())) {
                throw exception(CLASSIFICATION_NOT_EXISTS, classification.getName());
            }
        });
    }

    @Override
    public List<ClassificationDO> getSimpleClassifications(Collection<Long> ids) {
        return classificationMapper.selectBatchIds(ids);
    }

    @Override
    public Map<Long, ClassificationDO> getClassificationMap(Collection<Long> ids) {
        return ClassificationService.super.getClassificationMap(ids);
    }

    @Override
    public List<ClassificationDO> getClassificationList(Collection<Long> ids) {
        return classificationMapper.selectBatchIds(ids);
    }

    @Override
    public PageResult<ClassificationDO> getClassificationPage(ClassificationPageReqVO pageReqVO) {
        return classificationMapper.selectPage(pageReqVO);
    }

//    @Override
//    public List<ClassificationDO> getClassificationList(ClassificationExportReqVO exportReqVO) {
//        return classificationMapper.selectList(exportReqVO);
//    }

    @Override
    public List<ClassificationDO> getClassificationsByParentIdFromCache(Long parentId, boolean recursive) {
        if (parentId == null) {
            return Collections.emptyList();
        }
        List<ClassificationDO> result = new ArrayList<>(); // TODO 芋艿：待优化，新增缓存，避免每次遍历的计算
        // 递归，简单粗暴
        this.getClassificationsByParentIdFromCache(result, parentId,
                recursive ? Integer.MAX_VALUE : 1, // 如果递归获取，则无限；否则，只递归 1 次
                parentClassificationCache);
        return result;
    }

    /**
     * 递归获取所有的子部门，添加到 result 结果
     *
     * @param result 结果
     * @param parentId 父编号
     * @param recursiveCount 递归次数
     * @param parentClassificationMap 父部门 Map，使用缓存，避免变化
     */
    private void getClassificationsByParentIdFromCache(List<ClassificationDO> result, Long parentId, int recursiveCount,
                                                       Multimap<Long, ClassificationDO> parentClassificationMap) {
        // 递归次数为 0，结束！
        if (recursiveCount == 0) {
            return;
        }
        // 获得子部门
        Collection<ClassificationDO> classifications = parentClassificationMap.get(parentId);
        if (CollUtil.isEmpty(classifications)) {
            return;
        }
        result.addAll(classifications);
        // 继续递归
        classifications.forEach(classification -> getClassificationsByParentIdFromCache(result, classification.getId(),
                recursiveCount - 1, parentClassificationMap));
    }

    /**
     * 递归获取所有的子部门，添加到 result 结果
     *
     * @param result 结果
     * @param parentId 父编号
     * @param recursiveCount 递归次数
     * @param parentClassificationMap 父部门 Map，使用缓存，避免变化
     */
    private void getClassificationsByParentId(List<ClassificationDO> result, Long parentId, int recursiveCount,
                                              Multimap<Long, ClassificationDO> parentClassificationMap) {
        // 递归次数为 0，结束！
        if (recursiveCount == 0) {
            return;
        }
        // 获得子部门
        Collection<ClassificationDO> classifications = parentClassificationMap.get(parentId);
        if (CollUtil.isEmpty(classifications)) {
            return;
        }
        result.addAll(classifications);
        // 继续递归
        classifications.forEach(classification -> getClassificationsByParentId(result, classification.getId(),
                recursiveCount - 1, parentClassificationMap));
    }

}
